package com.vic.priorityqueue.confirm;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * 交换机确认的回调、消息未路由到队列的回调
 */
@Slf4j
@Component
public class MyConfirmCallback implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    // 依赖注入 rabbitTemplate 之后再设置它的回调对象
    @PostConstruct
    public void init() {
        // 注册 确认消息回调
        rabbitTemplate.setConfirmCallback(this);

        /**
         * true: 交换机无法将消息进行路由时，会将该消息返回给生产者，返回的消息可以从ReturnCallback回调中获取。
         * false: 如果发现消息无法进行路由，则直接丢弃。
         */
//        rabbitTemplate.setMandatory(false);

        // 注册 退回消息回调
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 交换机不管是否收到消息的一个回调方法
     * @param correlationData 消息相关数据
     * @param ack 交换机是否收到消息
     * @param cause 未收到消息的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData != null ? correlationData.getId() : "";
        if (ack) {
            log.info("交换机已经收到了 id 为 {} 的消息", id);
        } else {
            log.info("交换机没有收到了 id 为 {} 的消息，原因 {}", id, cause);
        }
    }

    /**
     * 如果交换机没有将消息路由到队列，会触发该回调方法
     * @param message
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.info("消息: {} 被服务器退回-->退回原因: {},交换机是: {},路由key是: {},退回编号是: {}",
                new String(message.getBody()), replyText, exchange, routingKey, replyCode);
    }
}
